Creating computation graphs in Tensor flow